import tensorflow as tf

a = tf.range(5)
b = tf.constant([2], dtype=tf.int32)

print('整除运算', a // b)
print('乘方运算', tf.pow(a, 3))
x = tf.cast(a, dtype=tf.float32)
print('平方运算', tf.square(x))
print('平方根', tf.sqrt(x))
print('指数运算', 2 ** x)
print('自然指数运算', tf.exp(2.))
# 其他对数运算要使用换地公式来计算
# 例如：tf.math.log(x)/tf.math.log(10.)  log𝑎x (以a为底x的对数)

# 在TensorFlow中使用@运算符可以方便实现矩阵相乘
A = tf.random.normal([2, 3])
B = tf.random.normal([3, 3])
print('矩阵相乘', A @ B)
